Fixed some missing statements