Check if we're always refer to the right layer

[LouisK92]

E.g. ssam refers to adata_sc.X - is this correct? Some other older codes were probably written without the pipeline's layers structure