Small improv

#324