Fixed bug/test