Minor fixes for lab scripts