I fixed a bug

Please accept my code!