Fixes bug based on @jperdereau comment (#325)

Thank you!