Fixed bug caused by test code which wasn't deleted.