Fixed bug - now compileable