Fixed so specs can be run properly