Tried to handle one-line case