Fixed bug left over from testing