Fixed code change mistake, overlooked this error