Fixed mistake according to comments.