minor fix following Benjamin Wickell tips