fix for issue #22 

Sorry I forgot to stage before I commited and pushed. 
Working with svn for work and got confused :(