REF: fixed gradientStep parsing
REF: minor changes
TEST: minor change