Minor script bugfix