Exercising Python3

Well today I started with a new tutorial using averages and I got stuck !
I did everything that the tutorial told me to do and type and I dont know why its just not working this is the tutorial code example: and this is the example of my code which for some reason it wont work.

count = 0
sum = 0.0
number = 1 
 
print("Enter 0 to exit the loop")
 
while number != 0:
    number = float(input("Enter a number: "))
    if number != 0:
        count = count + 1
        sum = sum + number
 
print("The average was:", sum / count)

and this is the example of my code which for some reason it wont work.
count = 0
sum = 0.0
number = 1

print("Enter 0 to exit the loop")
while number != 0:
    number = float(input("Enter a number: "))
            if number != 0:
               count = count + 1
               sum = sum + number
print("The average was: ", sum / count)