Fixed code

Changed needed to get the code to compile