Fixed error from teacher