minor fix for running correctly