Fixes an error with previous commit