Minor changes

Found minor bug in code i developed.