small program fix