Fixed bug for a noob mistake